#!/usr/bin/python3
# coding=utf-8
"""
寻找可以考虑投资的股票
1.通过company_info_category读取公司信息目录
2.对读取的公司信息（主要是财务分析）进行分析，找出近三年赢利、业务上涨的公司。
3.使用通达信下载前一天收盘数据
4.寻找策略:
  .近一年来从最高点跌入到最低点跌超过50%
  2.从最低点反弹10~20%后又下载到10%
  3.近三年业绩不错（没有负增长）
"""
from pytdx.reader import TdxDailyBarReader
reader = TdxDailyBarReader()
import os
MARKET_SZ_DATA_DIR="/media/lfh/data/greensoft/tdx/vipdoc/sz/lday"
MARKET_SH_DATA_DIR="/media/lfh/data/greensoft/tdx/vipdoc/sh/lday"

sz_codeLists=os.listdir(MARKET_SZ_DATA_DIR)
print("number code=",len(sz_codeLists))
#print(sz_codeLists)

for i in range(0,len(sz_codeLists)):
    filename=MARKET_SZ_DATA_DIR+"/"+sz_codeLists[i]
    print(filename)
    csvfile=sz_codeLists[i]
    #print(csvfile)
    code=csvfile[0:8]
    #print(code)
    df = reader.get_df(filename)
    df.to_csv("../data/xls/"+code+".csv")

#sh_codeLists=os.listdir(MARKET_SH_DATA_DIR)
#for i in range(0,len(sh_codeLists)):
#    print(sh_codeLists[i])
#print(sh_codeLists)

